/*
 * Copyright 2012 LinkedIn, Inc
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */

package com.linkedin.parseq.example.composite.classifier;

/**
 * @author Chris Pettitt (cpettitt@linkedin.com)
 */
public class Network {
  public static enum Distance {
    NOT_IN_NETWORK,
    SELF,
    D1,
    D2,
    D3
  }

  private final long memberId;

  public Network(long memberId) {
    this.memberId = memberId;
  }

  public Distance getDistance(final long memberId) {
    if (memberId == this.memberId)
      return Distance.SELF;

    final long group = memberId % 10;
    switch ((int) group) {
      case 10:
        return Distance.D3;
      case 9:
        return Distance.D2;
      case 8:
        return Distance.D1;
      default:
        return Distance.NOT_IN_NETWORK;
    }
  }
}
